from ai.llm.llm_chain import (testQuery, boatNormQuery)
from app import app
from flask import Flask, request


@app.route("/test")
def test():
    return f"test"


@app.post("/chat/test")
def chatTest():
    query = request.form.get("query")
    return testQuery(query)


@app.post("/chat/boatNorm")
def chatBoatNorm():
    query = request.form.get("query")
    return boatNormQuery(query)


if __name__ == '__main__':
    print("main test")
